import unittest
# from 第10天 import test_demo01, test_demo02, test_demo03, test_demo04
import unittestreport

if __name__ == '__main__':
    # 创建测试套件，测试套件加载当前目录下所有以test开头的.py文件
    my_suite = unittest.TestLoader().discover('.\作业', "test*.py")
    my_runner = unittestreport.TestRunner(
        suite=my_suite,
        filename="坚果的测试报告.html",
        report_dir="./reports",
        title='坚果的测试报告标题',
        tester='坚果',
        desc="unittest测试项目测试生成的报告",
        templates=2
    )
    # 运行,count=2表示测试用例执行失败重运行2次, interval=5表示重运行间隔时长5秒, thread_count=4
    my_runner.run(count=2, interval=1, thread_count=3)

# if __name__ == '__main__':
#     # 创建测试套件
#     my_suite = unittest.TestSuite()
#     # 创建加载器
#     my_loader = unittest.TestLoader()
#     # 使用加载器往测试套件中加载测试用例
#     my_suite.addTests([my_loader.loadTestsFromModule(test_demo01),
#                        my_loader.loadTestsFromModule(test_demo02),
#                        my_loader.loadTestsFromModule(test_demo03),
#                        my_loader.loadTestsFromModule(test_demo04)])
#     # 使用testRunner运行测试用例
#     unittest.TextTestRunner(verbosity=2).run(my_suite)